#include <gtest/gtest.h>
#include "DBManager/Database/DatabaseFactory.h"
#include "DBManager/DBServiceFactory.h"
#include "ElementCalculatorTest.hpp"

using namespace SG::DBManager;

/// @brief 数据库服务工厂
std::shared_ptr<DBServiceFactory> DB_SERVICEFACTORY = nullptr;

class H5DBEnvironment : public testing::Environment
{
public:
    DatabaseFactory* m_dbFactory = nullptr;

    void SetUp () override
    {
        // 数据库工厂
        m_dbFactory = new DatabaseFactory ();
        // 数据库
        IDatabase* db = m_dbFactory->GetDatabase (DatabaseFactory::HDF5, ".", "ElementUnitTest.db");
        // 根据数据库创建Service管理器
        DB_SERVICEFACTORY = std::make_shared<DBServiceFactory> (db);
    }
    void TearDown () override
    {
        // 释放数据库资源
        DB_SERVICEFACTORY.reset ();
        delete m_dbFactory;
    }
};

int main (int argc, char* argv[])
{
    testing::AddGlobalTestEnvironment (new H5DBEnvironment ());
    testing::InitGoogleTest (&argc, argv);
    return RUN_ALL_TESTS ();
}